from docx import Document
from docxcompose.composer import Composer
import os

def combine_docx_line(master, sub):
    if not os.path.exists(sub):#待合并文件必须存在
        return False

    if not master.endswith('.docx'):#主文件必须是docx格式（可以不存在）
        return False

    sub_docx = sub


    if os.path.exists(master):
        doc_master = Document(master)
        doc_master.add_paragraph()
        cp = Composer(doc_master)
        cp.append(Document(sub_docx))
    else:
        #master不存在，则sub直接给master
        doc_master = Document(sub_docx)

    doc_master.save(master)
    return True


def combine_docx_page(master, sub):
    if not os.path.exists(sub):#待合并文件必须存在
        return False

    if not master.endswith('.docx'):#主文件必须是docx格式（可以不存在）
        return False

    sub_docx = sub


    if os.path.exists(master):
        doc_master = Document(master)
        doc_master.add_page_break()
        cp = Composer(doc_master)
        cp.append(Document(sub_docx))
    else:
        #master不存在，则sub直接给master
        doc_master = Document(sub_docx)

    doc_master.save(master)
    return True
